﻿using NPOI.HPSF;
using NPOI.HSSF.UserModel;
using NPOI.SS.UserModel;
using System;
using System.Collections.Generic;
using System.Data;
using System.IO;
using System.Linq;
using System.Reflection;
using System.Text;
using System.Web;

namespace Common
{
    public class ExcelHelper
    {
        /// <summary>
        /// DataTable导出到Excel文件,v2版本
        /// </summary>
        /// <param name="dtSource">源DataTable</param>
        /// <param name="strHeaderText">表头文本</param>
        /// <param name="headers">需要导出的列的列头</param>
        /// <param name="cellKes">需要导出的对应的列字段</param>
        /// <param name="strFileName">保存位置</param>
        public static void Export(DataTable dtSource, string strHeaderText, string[] headers, string[] cellKes,
            string strFileName, string dict, string title = "")
        {
            if (!Directory.Exists(dict)) Directory.CreateDirectory(dict);
            string path = dict + strFileName;
            if (string.IsNullOrEmpty(title))
            {
                title = strHeaderText;
            }
            // 将需要导出的数据导到excel中并生成文件流
            using (MemoryStream ms = ExportToStream(dtSource, strHeaderText, headers, cellKes, title))
            {
                // 将文件流写入文件
                using (FileStream fs = new FileStream(path, FileMode.Create, FileAccess.Write))
                {
                    byte[] data = ms.ToArray();
                    fs.Write(data, 0, data.Length);
                    fs.Flush();
                }
            }
        }

        /// <summary>
        /// DataSet导出到Excel文件， v1版本，有空再将东西改成上面那个v2版本
        /// </summary>
        /// <param name="dsSource">源DataTable</param>
        /// <param name="strHeaderText">表头文本</param>
        /// <param name="headers">需要导出的列的列头</param>
        /// <param name="cellKes">需要导出的对应的列字段</param>
        /// <param name="strFileName">保存位置</param>
        public static void Export(DataSet dsSource, string[] strHeaderText, string[][] headers, string[][] cellKes, string strFileName, string dict)
        {
            if (!Directory.Exists(dict)) Directory.CreateDirectory(dict);
            string path = dict + strFileName;
            // 将需要导出的数据导到excel中并生成文件流
            using (MemoryStream ms = ExportToStream(dsSource, strHeaderText, headers, cellKes))
            {
                // 将文件流写入文件
                using (FileStream fs = new FileStream(path, FileMode.Create, FileAccess.Write))
                {
                    byte[] data = ms.ToArray();
                    fs.Write(data, 0, data.Length);
                    fs.Flush();
                }
            }
        }

        /// <summary>
        /// DataTable导出到Excel的MemoryStream
        /// </summary>
        /// <param name="dtSource">源DataTable</param>
        /// <param name="strHeaderText">表头文本</param>
        /// <param name="headers">需要导出的列的列头</param>
        /// <param name="cellKes">需要导出的对应的列字段</param>
        /// <returns></returns>
        private static MemoryStream ExportToStream(DataTable dtSource, string strHeaderText, string[] headers, string[] cellKes, string title)
        {
            // excel
            HSSFWorkbook workbook = new HSSFWorkbook();
            // 创建一个sheet页，已strHeaderText命名
            ISheet sheet = workbook.CreateSheet(strHeaderText);

            #region 右击文件 属性信息
            {
                DocumentSummaryInformation dsi = PropertySetFactory.CreateDocumentSummaryInformation();
                dsi.Company = "NPOI";
                workbook.DocumentSummaryInformation = dsi;

                SummaryInformation si = PropertySetFactory.CreateSummaryInformation();
                //si.Author = "文件作者信息"; //填加xls文件作者信息
                //si.ApplicationName = "创建程序信息"; //填加xls文件创建程序信息
                //si.LastAuthor = "最后保存者信息"; //填加xls文件最后保存者信息
                //si.Comments = "作者信息"; //填加xls文件作者信息
                //si.Title = "标题信息"; //填加xls文件标题信息
                //si.Subject = "主题信息";//填加文件主题信息
                si.CreateDateTime = DateTime.Now;
                workbook.SummaryInformation = si;
            }
            #endregion

            // 日期的样式
            ICellStyle dateStyle = workbook.CreateCellStyle();
            // 日期的格式化
            IDataFormat format = workbook.CreateDataFormat();
            dateStyle.DataFormat = format.GetFormat("yyyy-mm-dd");
            // 字体样式
            IFont datafont = workbook.CreateFont();
            // 字体大小
            datafont.FontHeightInPoints = 11;
            dateStyle.SetFont(datafont);
            // 边框
            dateStyle.BorderBottom = NPOI.SS.UserModel.BorderStyle.Thin;
            dateStyle.BorderLeft = NPOI.SS.UserModel.BorderStyle.Thin;
            dateStyle.BorderRight = NPOI.SS.UserModel.BorderStyle.Thin;
            dateStyle.BorderTop = NPOI.SS.UserModel.BorderStyle.Thin;

            // 其他数据的样式
            ICellStyle cellStyle = workbook.CreateCellStyle();
            // 字体样式
            IFont cellfont = workbook.CreateFont();
            // 字体大小
            cellfont.FontHeightInPoints = 11;
            cellStyle.SetFont(cellfont);
            // 边框
            cellStyle.BorderBottom = NPOI.SS.UserModel.BorderStyle.Thin;
            cellStyle.BorderLeft = NPOI.SS.UserModel.BorderStyle.Thin;
            cellStyle.BorderRight = NPOI.SS.UserModel.BorderStyle.Thin;
            cellStyle.BorderTop = NPOI.SS.UserModel.BorderStyle.Thin;

            // 总的列数
            int colNum = headers.Length;
            // 每个列的宽度
            int[] arrColWidth = new int[colNum];
            // 初始化列的宽度为列头的长度，已需要显示的列头的名字长度计算
            for (int i = 0; i < headers.Length; i++)
            {
                arrColWidth[i] = Encoding.GetEncoding(65001).GetBytes(headers[i]).Length;
            }

            // 循环数据，取每列数据最宽的作为该列的宽度
            for (int i = 0; i < dtSource.Rows.Count; i++)
            {
                for (int j = 0; j < cellKes.Length; j++)
                {
                    int intTemp = Encoding.GetEncoding(65001).GetBytes(dtSource.Rows[i][cellKes[j]].ToString()).Length;
                    if (intTemp > arrColWidth[j])
                    {
                        arrColWidth[j] = intTemp;
                    }
                }
            }

            // 记录生成的行数
            int rowIndex = 0;

            // DataTable中的列信息
            DataColumnCollection columns = dtSource.Columns;
            // 循环所有的行，向sheet页中添加数据
            foreach (DataRow row in dtSource.Rows)
            {
                #region 新建表，填充表头，填充列头，样式
                if (rowIndex == 65535 || rowIndex == 0)
                {
                    // 如果不是第一行，则创建一个新的sheet页
                    if (rowIndex != 0)
                    {
                        sheet = workbook.CreateSheet();
                    }

                    #region 表头及样式
                    {
                        // 在当前sheet页上创建第一行
                        IRow headerRow = sheet.CreateRow(0);
                        // 该行的高度
                        headerRow.HeightInPoints = 50;

                        // 设置第一列的值
                        headerRow.CreateCell(0).SetCellValue(title);

                        // 设置列的样式
                        ICellStyle headStyle = workbook.CreateCellStyle();

                        // 自动换行
                        headStyle.WrapText = true;
                        // 内容居中显示
                        headStyle.Alignment = HorizontalAlignment.Center;
                        // 字体样式
                        IFont font = workbook.CreateFont();
                        // 字体大小
                        font.FontHeightInPoints = 20;
                        // 粗体显示
                        //font.Boldweight = 700;
                        // 字体颜色
                        font.Color = NPOI.HSSF.Util.HSSFColor.Black.Index;

                        headStyle.SetFont(font);
                        // 边框
                        headStyle.BorderBottom = NPOI.SS.UserModel.BorderStyle.Thin;
                        headStyle.BorderLeft = NPOI.SS.UserModel.BorderStyle.Thin;
                        headStyle.BorderRight = NPOI.SS.UserModel.BorderStyle.Thin;
                        headStyle.BorderTop = NPOI.SS.UserModel.BorderStyle.Thin;

                        // 设置单元格的样式
                        headerRow.GetCell(0).CellStyle = headStyle;

                        // 设置该行每个单元格的样式
                        for (int i = 1; i < colNum; i++)
                        {
                            headerRow.CreateCell(i);
                            headerRow.GetCell(i).CellStyle = headStyle;
                        }

                        // 合并单元格
                        sheet.AddMergedRegion(new NPOI.SS.Util.CellRangeAddress(0, 0, 0, colNum - 1));
                    }
                    #endregion


                    #region 列头及样式
                    {
                        // 创建第二行
                        IRow headerRow = sheet.CreateRow(1);
                        // 该行的高度
                        headerRow.HeightInPoints = 28;
                        // 列的样式
                        ICellStyle headStyle = workbook.CreateCellStyle();
                        // 单元格内容居中显示
                        headStyle.Alignment = HorizontalAlignment.Center;
                        // 字体样式
                        IFont font = workbook.CreateFont();
                        // 字体大小
                        font.FontHeightInPoints = 12;
                        // 粗体
                        //font.Boldweight = 700;
                        // 字体颜色
                        font.Color = NPOI.HSSF.Util.HSSFColor.Black.Index;
                        headStyle.SetFont(font);
                        // 边框
                        headStyle.BorderBottom = NPOI.SS.UserModel.BorderStyle.Thin;
                        headStyle.BorderLeft = NPOI.SS.UserModel.BorderStyle.Thin;
                        headStyle.BorderRight = NPOI.SS.UserModel.BorderStyle.Thin;
                        headStyle.BorderTop = NPOI.SS.UserModel.BorderStyle.Thin;

                        // 设置每列的样式和值
                        for (int i = 0; i < headers.Length; i++)
                        {
                            headerRow.CreateCell(i).SetCellValue(headers[i]);
                            headerRow.GetCell(i).CellStyle = headStyle;
                            //if (arrColWidth[i] < 255 * 256)
                            //{
                            //    sheet.SetColumnWidth(i, arrColWidth[i] < 3000 ? 3000 : arrColWidth[i]);
                            //}
                            //else
                            //{
                            //    sheet.SetColumnWidth(i, 6000);
                            //}
                            ////设置列宽
                            sheet.SetColumnWidth(i, (arrColWidth[i]) * 256);
                        }
                    }
                    #endregion

                    rowIndex = 2;
                }
                #endregion


                #region 填充内容
                // 创建新的一行
                IRow dataRow = sheet.CreateRow(rowIndex);
                // 该行的高度
                dataRow.HeightInPoints = 23;
                // 循环需要写入的每列数据
                for (int i = 0; i < cellKes.Length; i++)
                {
                    // 创建列
                    ICell newCell = dataRow.CreateCell(i);
                    // 获取DataTable中该列对象
                    DataColumn column = columns[cellKes[i]];
                    // 该列的值
                    string drValue = row[column].ToString();

                    // 根据值得类型分别处理之后赋值
                    switch (column.DataType.ToString())
                    {
                        case "System.String"://字符串类型
                            newCell.SetCellValue(drValue);

                            newCell.CellStyle = cellStyle;
                            break;
                        case "System.DateTime"://日期类型
                            DateTime dateV;
                            DateTime.TryParse(drValue, out dateV);
                            newCell.SetCellValue(dateV);

                            newCell.CellStyle = dateStyle;//格式化显示
                            break;
                        case "System.Boolean"://布尔型
                            bool boolV = false;
                            bool.TryParse(drValue, out boolV);
                            newCell.SetCellValue(boolV);

                            newCell.CellStyle = cellStyle;
                            break;
                        case "System.Int16"://整型
                        case "System.Int32":
                        case "System.Int64":
                        case "System.Byte":
                            int intV = 0;
                            int.TryParse(drValue, out intV);
                            newCell.SetCellValue(intV);

                            newCell.CellStyle = cellStyle;
                            break;
                        case "System.Decimal"://浮点型
                        case "System.Double":
                            double doubV = 0;
                            double.TryParse(drValue, out doubV);
                            newCell.SetCellValue(doubV);

                            newCell.CellStyle = cellStyle;
                            break;
                        case "System.DBNull"://空值处理
                            newCell.SetCellValue("");

                            newCell.CellStyle = cellStyle;
                            break;
                        default:
                            newCell.SetCellValue("");

                            newCell.CellStyle = cellStyle;
                            break;
                    }
                }
                #endregion

                rowIndex++;
            }

            MemoryStream ms = new MemoryStream();
            workbook.Write(ms);
            ms.Flush();
            ms.Position = 0;
            return ms;
        }
        /// <summary>
        /// DataSet导出到Excel的MemoryStream
        /// </summary>
        /// <param name="dsSource">源DataTable</param>
        /// <param name="strHeaderText">表头文本</param>
        /// <param name="headers">需要导出的列的列头</param>
        /// <param name="cellKes">需要导出的对应的列字段</param>
        /// <returns></returns>
        private static MemoryStream ExportToStream(DataSet dsSource, string[] strHeaderText, string[][] headers, string[][] cellKes)
        {
            // excel
            HSSFWorkbook workbook = new HSSFWorkbook();
            // 创建一个sheet页，已strHeaderText命名

            #region 右击文件 属性信息
            {
                DocumentSummaryInformation dsi = PropertySetFactory.CreateDocumentSummaryInformation();
                dsi.Company = "NPOI";
                workbook.DocumentSummaryInformation = dsi;

                SummaryInformation si = PropertySetFactory.CreateSummaryInformation();
                //si.Author = "文件作者信息"; //填加xls文件作者信息
                //si.ApplicationName = "创建程序信息"; //填加xls文件创建程序信息
                //si.LastAuthor = "最后保存者信息"; //填加xls文件最后保存者信息
                //si.Comments = "作者信息"; //填加xls文件作者信息
                //si.Title = "标题信息"; //填加xls文件标题信息
                //si.Subject = "主题信息";//填加文件主题信息
                si.CreateDateTime = DateTime.Now;
                workbook.SummaryInformation = si;
            }
            #endregion

            // 日期的样式
            ICellStyle dateStyle = workbook.CreateCellStyle();
            // 日期的格式化
            IDataFormat format = workbook.CreateDataFormat();
            dateStyle.DataFormat = format.GetFormat("yyyy-mm-dd");
            // 字体样式
            IFont datafont = workbook.CreateFont();
            // 字体大小
            datafont.FontHeightInPoints = 11;
            dateStyle.SetFont(datafont);
            // 边框
            dateStyle.BorderBottom = NPOI.SS.UserModel.BorderStyle.Thin;
            dateStyle.BorderLeft = NPOI.SS.UserModel.BorderStyle.Thin;
            dateStyle.BorderRight = NPOI.SS.UserModel.BorderStyle.Thin;
            dateStyle.BorderTop = NPOI.SS.UserModel.BorderStyle.Thin;

            // 其他数据的样式
            ICellStyle cellStyle = workbook.CreateCellStyle();
            // 字体样式
            IFont cellfont = workbook.CreateFont();
            // 字体大小
            cellfont.FontHeightInPoints = 11;
            cellStyle.SetFont(cellfont);
            // 边框
            cellStyle.BorderBottom = NPOI.SS.UserModel.BorderStyle.Thin;
            cellStyle.BorderLeft = NPOI.SS.UserModel.BorderStyle.Thin;
            cellStyle.BorderRight = NPOI.SS.UserModel.BorderStyle.Thin;
            cellStyle.BorderTop = NPOI.SS.UserModel.BorderStyle.Thin;

            int sn = 0;
            foreach (DataTable dtSource in dsSource.Tables)
            {
                // 总的列数
                int colNum = headers[sn].Length;
                // 每个列的宽度
                int[] arrColWidth = new int[colNum];
                // 初始化列的宽度为列头的长度，已需要显示的列头的名字长度计算
                for (int i = 0; i < headers[sn].Length; i++)
                {
                    arrColWidth[i] = Encoding.GetEncoding(65001).GetBytes(headers[sn][i]).Length * 3;
                }


                ISheet sheet = workbook.CreateSheet(strHeaderText[sn]);
                // 循环数据，取每列数据最宽的作为该列的宽度
                for (int i = 0; i < dtSource.Rows.Count; i++)
                {
                    for (int j = 0; j < cellKes[sn].Length; j++)
                    {
                        int intTemp = Encoding.GetEncoding(65001).GetBytes(dtSource.Rows[i][cellKes[sn][j]].ToString()).Length;
                        if (intTemp > arrColWidth[j])
                        {
                            arrColWidth[j] = intTemp;
                        }
                    }
                }

                // 记录生成的行数
                int rowIndex = 0;

                // DataTable中的列信息
                DataColumnCollection columns = dtSource.Columns;
                // 循环所有的行，向sheet页中添加数据
                foreach (DataRow row in dtSource.Rows)
                {
                    #region 新建表，填充表头，填充列头，样式
                    if (rowIndex == 65535 || rowIndex == 0)
                    {
                        // 如果不是第一行，则创建一个新的sheet页
                        if (rowIndex != 0)
                        {
                            sheet = workbook.CreateSheet();
                        }

                        #region 表头及样式
                        {
                            // 在当前sheet页上创建第一行
                            IRow headerRow = sheet.CreateRow(0);
                            // 该行的高度
                            headerRow.HeightInPoints = 50;
                            // 设置第一列的值
                            headerRow.CreateCell(0).SetCellValue(strHeaderText[sn]);

                            // 设置列的样式
                            ICellStyle headStyle = workbook.CreateCellStyle();
                            // 内容居中显示
                            headStyle.Alignment = HorizontalAlignment.Center;
                            // 字体样式
                            IFont font = workbook.CreateFont();
                            // 字体大小
                            font.FontHeightInPoints = 20;
                            // 粗体显示
                            //font.Boldweight = 700;
                            // 字体颜色
                            font.Color = NPOI.HSSF.Util.HSSFColor.Black.Index;

                            headStyle.SetFont(font);
                            // 边框
                            headStyle.BorderBottom = NPOI.SS.UserModel.BorderStyle.Thin;
                            headStyle.BorderLeft = NPOI.SS.UserModel.BorderStyle.Thin;
                            headStyle.BorderRight = NPOI.SS.UserModel.BorderStyle.Thin;
                            headStyle.BorderTop = NPOI.SS.UserModel.BorderStyle.Thin;

                            // 设置单元格的样式
                            headerRow.GetCell(0).CellStyle = headStyle;

                            // 设置该行每个单元格的样式
                            for (int i = 1; i < colNum; i++)
                            {
                                headerRow.CreateCell(i);
                                headerRow.GetCell(i).CellStyle = headStyle;
                            }

                            // 合并单元格
                            sheet.AddMergedRegion(new NPOI.SS.Util.CellRangeAddress(0, 0, 0, colNum - 1));
                        }
                        #endregion


                        #region 列头及样式
                        {
                            // 创建第二行
                            IRow headerRow = sheet.CreateRow(1);
                            // 该行的高度
                            headerRow.HeightInPoints = 28;
                            // 列的样式
                            ICellStyle headStyle = workbook.CreateCellStyle();
                            // 单元格内容居中显示
                            headStyle.Alignment = HorizontalAlignment.Center;
                            // 字体样式
                            IFont font = workbook.CreateFont();
                            // 字体大小
                            font.FontHeightInPoints = 12;
                            // 粗体
                            //font.Boldweight = 700;
                            // 字体颜色
                            font.Color = NPOI.HSSF.Util.HSSFColor.Black.Index;
                            headStyle.SetFont(font);
                            // 边框
                            headStyle.BorderBottom = NPOI.SS.UserModel.BorderStyle.Thin;
                            headStyle.BorderLeft = NPOI.SS.UserModel.BorderStyle.Thin;
                            headStyle.BorderRight = NPOI.SS.UserModel.BorderStyle.Thin;
                            headStyle.BorderTop = NPOI.SS.UserModel.BorderStyle.Thin;

                            // 设置每列的样式和值
                            for (int i = 0; i < headers[sn].Length; i++)
                            {
                                headerRow.CreateCell(i).SetCellValue(headers[sn][i]);
                                headerRow.GetCell(i).CellStyle = headStyle;
                                //if (arrColWidth[i] < 255 * 256)
                                //{
                                //    sheet.SetColumnWidth(i, arrColWidth[i] < 3000 ? 3000 : arrColWidth[i]);
                                //}
                                //else
                                //{
                                //    sheet.SetColumnWidth(i, 6000);
                                //}
                                ////设置列宽
                                sheet.SetColumnWidth(i, (arrColWidth[i] + 1) * 256);
                            }
                        }
                        #endregion

                        rowIndex = 2;
                    }
                    #endregion


                    #region 填充内容
                    // 创建新的一行
                    IRow dataRow = sheet.CreateRow(rowIndex);
                    // 该行的高度
                    dataRow.HeightInPoints = 23;
                    // 循环需要写入的每列数据
                    for (int i = 0; i < cellKes[sn].Length; i++)
                    {
                        // 创建列
                        ICell newCell = dataRow.CreateCell(i);
                        // 获取DataTable中该列对象
                        DataColumn column = columns[cellKes[sn][i]];
                        // 该列的值
                        string drValue = row[column].ToString();

                        // 根据值得类型分别处理之后赋值
                        switch (column.DataType.ToString())
                        {
                            case "System.String"://字符串类型
                                newCell.SetCellValue(drValue);

                                newCell.CellStyle = cellStyle;
                                break;
                            case "System.DateTime"://日期类型
                                DateTime dateV;
                                DateTime.TryParse(drValue, out dateV);
                                newCell.SetCellValue(dateV);

                                newCell.CellStyle = dateStyle;//格式化显示
                                break;
                            case "System.Boolean"://布尔型
                                bool boolV = false;
                                bool.TryParse(drValue, out boolV);
                                newCell.SetCellValue(boolV);

                                newCell.CellStyle = cellStyle;
                                break;
                            case "System.Int16"://整型
                            case "System.Int32":
                            case "System.Int64":
                            case "System.Byte":
                                int intV = 0;
                                int.TryParse(drValue, out intV);
                                newCell.SetCellValue(intV);

                                newCell.CellStyle = cellStyle;
                                break;
                            case "System.Decimal"://浮点型
                            case "System.Double":
                                double doubV = 0;
                                double.TryParse(drValue, out doubV);
                                newCell.SetCellValue(doubV);

                                newCell.CellStyle = cellStyle;
                                break;
                            case "System.DBNull"://空值处理
                                newCell.SetCellValue("");

                                newCell.CellStyle = cellStyle;
                                break;
                            default:
                                newCell.SetCellValue("");

                                newCell.CellStyle = cellStyle;
                                break;
                        }
                    }
                    #endregion

                    rowIndex++;
                }
                sn++;
            }

            MemoryStream ms = new MemoryStream();
            workbook.Write(ms);
            ms.Flush();
            ms.Position = 0;
            return ms;
        }
    }
}